package com.sheng.leetcode.year2022.month09.day08;

import org.junit.Test;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * @author liusheng
 * @date 2022/09/08
 *
 * 667. 优美的排列 II
 *
 * 给你两个整数 n 和 k ，请你构造一个答案列表 answer ，该列表应当包含从 1 到 n 的 n 个不同正整数，并同时满足下述条件：
 * 假设该列表是 answer =[a1, a2, a3, ... , an] ，那么列表 [|a1 - a2|, |a2 - a3|, |a3 - a4|, ... , |an-1 - an|] 中应该有且仅有 k 个不同整数。
 * 返回列表 answer 。如果存在多种答案，只需返回其中 任意一种 。
 *
 * 示例 1：
 * 输入：n = 3, k = 1
 * 输出：[1, 2, 3]
 * 解释：[1, 2, 3] 包含 3 个范围在 1-3 的不同整数，并且 [1, 1] 中有且仅有 1 个不同整数：1
 *
 * 示例 2：
 * 输入：n = 3, k = 2
 * 输出：[1, 3, 2]
 * 解释：[1, 3, 2] 包含 3 个范围在 1-3 的不同整数，并且 [2, 1] 中有且仅有 2 个不同整数：1 和 2
 *
 * 提示：
 *
 * 1 <= k < n <= 104
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/beautiful-arrangement-ii
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
public class LeetCode0667 {

    @Test
    public void test01() {
//        int n = 3;
//        int k = 1;
//        int n = 3;
//        int k = 2;
//        int n = 4;
//        int k = 2;
        int n = 5;
        int k = 4;
        System.out.println(Arrays.toString(new Solution().constructArray(n, k)));
    }
}
class Solution {
    public int[] constructArray(int n, int k) {
        int[] answer = new int[n];
        // 对答案数组进行赋值
        for (int i = 0; i < n; i++) {
            answer[i] = i + 1;
        }
        if (k == 1) {
            return answer;
        }
        // 思路：根据数学规律，可以获得：
        // 一边从k + 1递减插入数组，一边从1递增开始插入数组，刚好满足题目要求
        // x进行递减
        int x = k + 1;
        // y进行递增
        int y = 1;
        // index为插入的索引
        int index = 0;
        // 根据k的奇偶情况进行分类讨论
        if (k % 2 != 0) {
            for (int i = 0; i <= k / 2; i++) {
                answer[index] = x;
                answer[index + 1] = y;
                index += 2;
                x--;
                y++;
            }
        } else {
            for (int i = 0; i < k / 2; i++) {
                answer[index] = x;
                answer[index + 1] = y;
                index += 2;
                x--;
                y++;
            }
            answer[index] = y;
        }
        return answer;
    }
}
// 超时
//class Solution {
//    int x;
//    int y;
//    int[] answer;
//    boolean flag = false;
//    // 记录整数间差值
//    List<Integer> list;
//    public int[] constructArray(int n, int k) {
//        answer = new int[n];
//        // 同样使用dfs，题目条件，使用1-n的n个不同整数，并且满足两两之间的差值个数要等于k
//        // ints代表整数是否被使用
//        int[] ints = new int[n];
//        x = n;
//        y = k;
//        for (int i = 1; i <= x; i++) {
//            ints[i - 1] = 1;
//            answer[0] = i;
//            list = new ArrayList<>();
//            dfs(list, 2, ints);
//            if (flag) {
//                return answer;
//            }
//            ints[i - 1] = 0;
//            answer[0] = 0;
//        }
//        return new int[n];
//    }
//
//    public void dfs(List<Integer> list, int num, int[] ints) {
//        if (num > x) {
//            if (list.size() == y) {
//                flag = true;
//            }
//            return;
//        }
//        for (int i = 1; i <= x; i++) {
//            if (flag) {
//                break;
//            }
//            if (ints[i - 1] == 0) {
//                int abs = Math.abs(i - answer[num - 2]);
//                if (list.contains(abs)) {
//                    ints[i - 1] = 1;
//                    answer[num - 1] = i;
//                    dfs(list, num + 1, ints);
//                    if (!flag) {
//                        ints[i - 1] = 0;
//                        answer[num - 1] = 0;
//                    }
//                } else if (!list.contains(abs) && list.size() + 1 <= y) {
//                    list.add(abs);
//                    ints[i - 1] = 1;
//                    answer[num - 1] = i;
//                    dfs(list, num + 1, ints);
//                    if (!flag) {
//                        ints[i - 1] = 0;
//                        answer[num - 1] = 0;
//                        list.remove(list.size() - 1);
//                    }
//                }
//            }
//        }
//    }
//}
